home *** CD-ROM | disk | FTP | other *** search
/ Turnbull China Bikeride / Turnbull China Bikeride - Disc 2.iso / STUTTGART / UNIXTOOL / GNU / GNUDBM / GNUDBM~ / <tarZ$use> / files / GNUDbm / C / Gdbmreorg < prev    next >
Text File  |  1990-03-01  |  4KB  |  163 lines
  1. /* gdbmreorg.c - Reorganize the database file. */
  2.  
  3. /*  GNU DBM  - DataBase Manager (database subroutines) by Philip A. Nelson
  4.     Copyright (C) 1989  Free Software Foundation, Inc.
  5.  
  6.     This program is free software; you can redistribute it and/or modify
  7.     it under the terms of the GNU General Public License as published by
  8.     the Free Software Foundation.
  9.  
  10.     This program is distributed in the hope that it will be useful,
  11.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13.     GNU General Public License for more details.
  14.  
  15.     You should have received a copy of the GNU General Public License
  16.     along with this program; if not, write to the Free Software
  17.     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  18.  
  19.     You may contact the author by:
  20.        e-mail:  phil@wwu.edu
  21.       us-mail:  Philip A. Nelson
  22.                 Computer Science Department
  23.                 Western Washington University
  24.                 Bellingham, WA 98226
  25.         phone:  (206) 676-3035
  26.  
  27. ************************************************************************/
  28.  
  29. #include "gdbm.h"
  30. #include "extern.h"
  31. #include "gdbmerrno.h"
  32. #include "gdbmutils.h"
  33.  
  34. #include <stdio.h>
  35. #include <stdlib.h>
  36. #include <string.h>
  37.  
  38. /* Reorganize the database.  This requires creating a new file and inserting
  39.    all the elements in the old file DBF into the new file.  The new file
  40.    is then renamed to the same name as the old file and DBF is updated to
  41.    contain all the correct information about the new file. */
  42.  
  43. int gdbm_reorganize(dbm_file_info * dbf)
  44. {
  45.     dbm_file_info *new_dbf;
  46.     char *old_name;
  47.     char *new_name;
  48.     int len;
  49.     datum key, nextkey, data;
  50.  
  51.     /* Readers can not reorganize! */
  52.     if (dbf->read_write != DBM_WRITER)
  53.     {
  54.         gdbm_errno = READER_CANT_REORGANIZE;
  55.         return -1;
  56.     }
  57.  
  58.     /* Construct new name for temporary file. */
  59.     len = strlen(dbf->name);
  60.     old_name = (char *) malloc(len + 1);
  61.     new_name = (char *) malloc(len + 1);
  62.     if (old_name == NULL || new_name == NULL)
  63.     {
  64.         if (old_name != NULL)
  65.             free(old_name);
  66.         gdbm_errno = MALLOC_ERROR;
  67.         return -1;
  68.     }
  69.     strcpy(old_name, dbf->name);
  70.     strcpy(new_name, dbf->name);
  71.     new_name[len - 1] = '!';
  72.  
  73.     /* Touch new file and open with gdbm. */
  74.     fclose(fopen(new_name, "w"));
  75.  
  76.     new_dbf = gdbm_open(new_name, dbf->header.block_size, DBM_WRITER,
  77.                 dbf->fatal_err);
  78.  
  79.     if (new_dbf == NULL)
  80.     {
  81.         gdbm_errno = REORGANIZE_FAILED;
  82.         free(old_name);
  83.         free(new_name);
  84.         return -1;
  85.     }
  86.  
  87.     if (dbf->header.update_state == -1)
  88.         gdbm_recovery(new_dbf, FALSE);
  89.  
  90.     /* For each item in the old database, add an entry in the new. */
  91.     key = gdbm_firstkey(dbf);
  92.  
  93.     while (key.dptr != NULL)
  94.     {
  95.         data = gdbm_fetch(dbf, key);
  96.         if (data.dptr != NULL)
  97.         {
  98.             /* Add the data to the new file. */
  99.             gdbm_store(new_dbf, key, data);
  100.         }
  101.         else
  102.         {
  103.             /* ERROR! Abort and don't finish reorganize. */
  104.             gdbm_close(new_dbf);
  105.             gdbm_errno = REORGANIZE_FAILED;
  106.             remove(new_name);
  107.             free(old_name);
  108.             free(new_name);
  109.             return -1;
  110.         }
  111.         nextkey = gdbm_nextkey(dbf, key);
  112.         free(key.dptr);
  113.         key = nextkey;
  114.     }
  115.  
  116.     /* Remove the old file... we are now comitted */
  117.     fclose(dbf->desc);
  118.     remove(old_name);
  119.  
  120.     /* Move the new file to old name. */
  121.  
  122.     gdbm_close(new_dbf);
  123.     if (rename(new_name, dbf->name) != 0)
  124.     {
  125.         gdbm_errno = REORGANIZE_FAILED;
  126.         free(old_name);
  127.         free(new_name);
  128.         return -1;
  129.     }
  130.  
  131.     /* Fix up DBF to have the correct information for the new file. */
  132.     new_dbf = gdbm_open(dbf->name, dbf->header.block_size, DBM_WRITER,
  133.                 dbf->fatal_err);
  134.  
  135.     if (new_dbf == NULL)
  136.     {
  137.         gdbm_errno = REORGANIZE_FAILED;
  138.         free(old_name);
  139.         free(new_name);
  140.         return -1;
  141.     }
  142.     if (dbf->dir != NULL)
  143.         free(dbf->dir);
  144.     if (dbf->bucket != NULL)
  145.         free(dbf->bucket);
  146.     if (dbf->avail != NULL)
  147.         free(dbf->avail);
  148.     dbf->desc = new_dbf->desc;
  149.     dbf->header = new_dbf->header;
  150.     dbf->dir = new_dbf->dir;
  151.     dbf->bucket = new_dbf->bucket;
  152.     dbf->bucket_adr = new_dbf->bucket_adr;
  153.     dbf->bucket_dir = new_dbf->bucket_dir;
  154.     dbf->avail = new_dbf->avail;
  155.     dbf->avail_adr = new_dbf->avail_adr;
  156.     free(new_dbf->name);
  157.     free(new_dbf);
  158.     free(old_name);
  159.     free(new_name);
  160.  
  161.     return 0;
  162. }
  163.